/*
* Copyright (c) Huawei Technologies Co., Ltd. 2025-2025. All rights reserved.
* ubs-engine is licensed under Mulan PSL v2.
* You can use this software according to the terms and conditions of the Mulan PSL v2.
* You may obtain a copy of Mulan PSL v2 at:
*          http://license.coscl.org.cn/MulanPSL2
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PSL v2 for more details.
*/

#include "ubse_error.h"
#include "test_ubse_ras_module.h"

namespace ubse::ras::ut {
void TestUbseRasModule::SetUp()
{
    Test::SetUp();
}

void TestUbseRasModule::TearDown()
{
    Test::TearDown();
    GlobalMockObject::verify();
}

TEST_F(TestUbseRasModule, Initialize)
{
    auto module = std::make_shared<UbseRasModule>();
    auto res = module->Initialize();
    ASSERT_EQ(res, UBSE_OK);
}

TEST_F(TestUbseRasModule, StartWhenStartRasHandlerError)
{
    auto module = std::make_shared<UbseRasModule>();
    MOCKER_CPP(&UbseRasHandler::StartRasHandler).stubs().will(returnValue(UBSE_ERROR));
    auto res = module->Start();
    ASSERT_EQ(res, UBSE_ERROR);
}

TEST_F(TestUbseRasModule, StartSuccess)
{
    auto module = std::make_shared<UbseRasModule>();
    MOCKER_CPP(&UbseRasHandler::StartRasHandler).stubs().will(returnValue(UBSE_OK));
    auto res = module->Start();
    ASSERT_EQ(res, UBSE_OK);
}
}